#-*- coding: utf-8 -*-

class CLS_Modl_Values_Statistics():

	''' Model -> Plot das estatísticas '''
	
	def statistics_plot1(self, lista2, curso, unidade):
		
		''' Plot -> para cada curso selecionado, da evolução do número de alunos inscritos ao longo dos anos
		
		var lista2 - lista com os anos, unidades , cursos e o nr de alunos
		var curso - curso a procurarcsv
		var unidade - unidade escolhida
		return lista com lista de anos e lista com quantidade de alunos de acordo com o curso e unidade escolhida
		'''

		lista_anos = []
		lista_quantidade = []
			
		for uni_curso_lst in lista2:
			if (uni_curso_lst[1].decode('utf8') == unidade and uni_curso_lst[2].decode('utf8') == curso):
				lista_anos.append(uni_curso_lst[0])
				lista_quantidade.append(uni_curso_lst[3])
				
		return ([lista_anos, lista_quantidade])

	def statistics_plot2(self, lista_unidades, lista4, nivel, unidade):
		
		''' Plot -> evolução do número total de alunos inscritos ao longo dos anos por nível de formação
		var lista2 - lista com anos, uni, curso e nr alunos
		var listas4 - lista com os anos, niveis e quantidade de alunos
		var nivel - nivel escolhido
		var unidade - unidade escolhida
		return lista com lista de anos e quantidade de alunos consoante o nivel
		'''

		lista_anos = []
		lista_quantidade = []
		novo_nivel = 0

		for unidade_lst in lista_unidades:
			if(unidade_lst == unidade):
				for nivel_lst4 in lista4:
					if(nivel_lst4[1].decode('utf8') == nivel):
						if(novo_nivel == 0):		# not lista_anos and not lista_quantidade
							print 'ANOS : ', nivel_lst4[0]
							lista_anos.append(nivel_lst4[0])
							print 'QNT : ', nivel_lst4[2]
							lista_quantidade.append(nivel_lst4[2])
						#else:
						#	print 'ADDDD plot 2'
						#	index = lista_anos.index(nivel_lst4[0])
						#	nr = lista_quantidade[index] + nivel_lst4[2]
						#	lista_quantidade.insert(index, nr)
				#novo_nivel = 1
		return ([lista_anos, lista_quantidade])
